BATCH_SIZE = 8
train_data = NumberDataset(path="data/captcha/train",type_transform=my_transform)
test_data = NumberDataset(path="data/captcha/test",type_transform=my_transform)
# 训练集
train_loader = DataLoader(train_data,
                               batch_size=BATCH_SIZE,
                               shuffle=True,
                               drop_last=True)
# 测试集
test_loader = DataLoader(test_data,
                              batch_size=BATCH_SIZE,
                              shuffle=True,
                              drop_last=True)
